rainbow in python